//完全平⽅数（medium）： https://leetcode.cn/problems/perfect-squares/
class Solution
{
public:
	int numSquares(int n)
	{
		vector<int> dp(n + 1);
		dp[1] = 1; // 初始化
		for (int i = 2; i <= n; i++) // 枚举每个数
		{
			dp[i] = 1 + dp[i - 1]; // ⾄少等于 1 + dp[i - 1]
			for (int j = 2; j * j <= i; j++)
			{
				// ⽤⼩于 i 的完全平⽅数划分区间
				dp[i] = min(dp[i], dp[i - j * j] + 1); // 拿到所有划分区间内的最⼩
			}
			
		}
		// 返回结果
		return dp[n];
	}
};

//⼀和零（medium）: https://leetcode.cn/problems/ones-and-zeroes/
class Solution {
public:
	int findMaxForm(vector<string>& strs, int m, int n) {
		int len = strs.size();
		vector<vector<vector<int>>> dp(len + 1, vector<vector<int>>(m + 1,vector<int>(n + 1)));
		for (int i = 1; i <= len; i++)
		{
			// 统计⼀下 0 1 的个数
			int a = 0, b = 0;
			for (auto ch : strs[i - 1])
			{
				if (ch == '0') a++;
				else b++;
			}
				
			for (int j = m; j >= 0; j--)
			{
				for (int k = n; k >= 0; k--)
				{
					dp[i][j][k] = dp[i - 1][j][k];
					if (j >= a && k >= b)
					{
						dp[i][j][k] = max(dp[i][j][k], dp[i - 1][j - a][k - b]+ 1);
					}
						
				}
			}
				
		}
		return dp[len][m][n];
	}
};